package com.udf.hive.udf;

import org.apache.hadoop.hive.ql.exec.UDF;

public class IdNum2Sex extends UDF {
    public String evaluate(String input) {
        if (input != null) {
            String idNumTrim = input.trim();
            String sexCode = null;
            if (idNumTrim.length() == 15) {
                sexCode = input.substring(14, 15);
                if (Integer.parseInt(sexCode) % 2 == 0) {
                    return "女";
                } else {
                    return "男";
                }
            } else if (idNumTrim.length() == 18) {
                sexCode = input.substring(16, 17);
                if (Integer.parseInt(sexCode) % 2 == 0) {
                    return "女";
                } else {
                    return "男";
                }
            }
        }
        return "";
    }
}
